# -*- coding: utf-8 -*-
"""
Created on Thu Oct 20 21:16:19 2022

@author: nchji
"""
import numpy as np
import pandas as pd
import random
def gen1(data):
    l=[]
    for k,v in data.items():
        l=l+[k]*v
    random.shuffle(l)
    return l
def gen_normal(mean=3.5,n=588):
    l=[]
    for i in np.random.normal(mean, 2, [n]):
        if i<1:
            l.append(1)
        elif i>5:
            l.append(5)
        else:
            l.append(round(i))
    return l

data0=dict(
    info_sex=gen1({1:283,2:305}),
    info_age=gen1({1:61,2:254,3:209,4:54,5:10}),
    info_iscity=gen1({1:106,2:482}),
    info_phd=gen1({1:14,2:27,3:454,4:93}),
    info_income=gen1({1:4,2:20,3:273,4:291}),
    info_work=gen1({1:66,2:112,3:376,4:28,5:6}),
    info_freq=gen1({1:564,2:23,3:1}),
    info_freqfunc=gen1({1:540,2:30,3:15,4:3}),
    Q1=gen_normal(mean=3.9,n=588),
    Q2=gen_normal(mean=3.2,n=588),
    Q3=gen_normal(mean=3.3,n=588),
    Q4=gen_normal(mean=3.1,n=588),
    Q5=gen_normal(mean=3.1,n=588),
    Q6=gen_normal(mean=3.0,n=588),
    Q7=gen_normal(mean=3.0,n=588),
    Q8=gen_normal(mean=2.7,n=588),
    Q9=gen_normal(mean=3.5,n=588),
    Q10=gen_normal(mean=2.9,n=588),
    Q11=gen_normal(mean=3.0,n=588),
    Q12=gen_normal(mean=3.0,n=588),
    Q13=gen_normal(mean=4.0,n=588),
    Q14=gen_normal(mean=3.2,n=588),
    Q15=gen_normal(mean=3.3,n=588),
    Q16=gen_normal(mean=4.0,n=588),
    Q17=gen_normal(mean=3.5,n=588),
    Q18=gen_normal(mean=3.1,n=588),
    Q19=gen_normal(mean=3.5,n=588),
    Q20=gen_normal(mean=3.5,n=588),
    Q21=gen_normal(mean=2.8,n=588),
    Q22=gen_normal(mean=3.8,n=588),
    Q23=gen_normal(mean=2.1,n=588),
    Q24=gen_normal(mean=4.2,n=588),
    Q25=gen_normal(mean=3.5,n=588),
    Q26=gen_normal(mean=3.5,n=588),
    Q27=gen_normal(mean=3.5,n=588),
    Q28=gen_normal(mean=4.1,n=588),
    Q29=gen_normal(mean=3.5,n=588),
    Q30=gen_normal(mean=4.3,n=588),
    Q31=gen_normal(mean=3.5,n=588),
    Q32=gen_normal(mean=4.1,n=588)
    )

data=pd.DataFrame(data0)
data.to_csv("data.csv")